import java.util.Arrays;

public class Solution416 {
    /**
     * 给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集，使得两个子集的元素和相等。
     * <p>
     * 注意:
     * <p>
     * 每个数组中的元素不会超过 100
     * 数组的大小不会超过 200
     * 示例 1:
     * <p>
     * 输入: [1, 5, 11, 5]
     * <p>
     * 输出: true
     * <p>
     * 解释: 数组可以分割成 [1, 5, 5] 和 [11].
     */

    //深搜
    public boolean canPartition(int[] nums) {

        Arrays.sort(nums);
        int sum = 0;
        for (int num : nums) {
            sum += num;
        }
        if ((sum & 1) == 1) {
            return false;
        }
        sum >>= 1;
        return canPartition(nums, nums.length - 1, sum, sum);
    }

    public boolean canPartition(int[] nums, int idx, int part1, int part2) {
        if (part1 == 0 || part2 == 0) {
            return true;
        } else if (part1 < 0 || part2 < 0) {
            return false;
        } else {
            return canPartition(nums, idx - 1, part1 - nums[idx], part2) ||
                    canPartition(nums, idx - 1, part1, part2 - nums[idx]);
        }
    }

    // 动态规划 0-1背包
    public boolean canPartition2(int[] nums) {
        int len = nums.length;
        if (len == 0) {
            return false;
        }

        int sum = 0;
        for (int num : nums) {
            sum += num;
        }

        if ((sum & 1) == 1) {
            return false;
        }

        int target = sum / 2;

        boolean[] dp = new boolean[target + 1];
        dp[0] = true;

        if (nums[0] <= target) {
            dp[nums[0]] = true;
        }

        for (int i = 1; i < len; i++) {
            for (int j = target; nums[i] <= j; j--) {
                if (dp[target]) {
                    return true;
                }

                dp[j] = dp[j] || dp[j - nums[i]];
            }
        }
        return dp[target];
    }
}
